package com.crm.miaohe.utils;

import org.mybatis.generator.api.MyBatisGenerator;
import org.mybatis.generator.config.Configuration;
import org.mybatis.generator.config.xml.ConfigurationParser;
import org.mybatis.generator.internal.DefaultShellCallback;

import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;

public class MyBatisGeneratorRunner {
	public static void main(String[] args) {
        try {
            // 警告信息列表
            List<String> warnings = new ArrayList<>();
            
            // 是否覆盖已有文件
            boolean overwrite = true;
            
            // 1. 加载配置文件
            InputStream inputStream = MyBatisGeneratorRunner.class
                .getClassLoader()
                .getResourceAsStream("generatorConfig.xml");
            
            // 2. 解析配置文件
            ConfigurationParser cp = new ConfigurationParser(warnings);
            Configuration config = cp.parseConfiguration(inputStream);
            inputStream.close();
            
            // 3. 创建回调对象
            DefaultShellCallback callback = new DefaultShellCallback(overwrite);
            
            // 4. 创建生成器
            MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings);
            
            // 5. 执行生成
            myBatisGenerator.generate(null);
            
            // 6. 输出警告信息
            for (String warning : warnings) {
                System.out.println("Warning: " + warning);
            }
            
            System.out.println("MyBatis Generator 代码生成完成!");
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}
